Efficient coding in Python